package com.example.mq.mqserver.datacenter;

import com.example.mq.MqApplication;
import com.example.mq.mqserver.core.Binding;
import com.example.mq.mqserver.core.Exchange;
import com.example.mq.mqserver.core.ExchangeType;
import com.example.mq.mqserver.core.MSGQueue;
import com.example.mq.mqserver.mapper.MetaMapper;
import org.springframework.beans.factory.annotation.Autowired;

import java.io.File;
import java.util.List;

/**
 * 通过这个类来整合数据库的操作
 */
public class DataBaseManager {


    private MetaMapper metaMapper;

    //数据库的初始化=建库建表+插入一些默认数据
    //我们期望,在咱们的 broker server 启动的时候,做出下列逻辑判定:
    // 1.如果数据库已经存在了，(表啥的都有了),不做任何操作.(通过 meta.db判断是否存在）
    // 2.如果数据库不存在,则创建库,创建表,构造默认数据
    public void init(){
        //手动的获取到MetaMapper
        metaMapper= MqApplication.context.getBean(MetaMapper.class);
        if(!checkDBExists()){
            //如果数据库不存在，建库建表
            //先创建一个data目录
            File dataDir=new File("./data");
            dataDir.mkdirs();
            //建表
            createTable();
            //插入默认数据
            createDefaultData();
            System.out.println("[DataBaseManager]数据库初始化完成！");
        }else {
            System.out.println("[DataBaseManager]数据库已经存在！");
        }

    }
    public void deleteDB(){
        File file=new File("./data/meta.db");
        boolean ret=file.delete();
        if(ret){
            System.out.println("[DataBaseManager]删除数据库成功！");
        }else{
            System.out.println("[DataBaseManager]删除数据库失败！");
        }
        //删除之前保证这个目录里面是空的
        File dataDir=new File("./data");
        ret = dataDir.delete();
        if(ret){
            System.out.println("[DataBaseManager]删除数据库目录成功！");
        }else{
            System.out.println("[DataBaseManager]删除数据库目录失败！");
        }

    }

    private boolean checkDBExists() {
        //通过 meta.db判断是否存在
        File file=new File("./data/meta.db");
        if(file.exists()){
            return true;
        }
        return false;
    }

    //这个方法用来建表
    //建库操作并不需要手动执行（不需要手动创建meta.db文件）
    //首次执行这里的数据库操作的时候，就会自动创建出meta.db文件来（mybaits帮我们完成的）
    private void createTable() {
        metaMapper.createExchangeTable();
        metaMapper.createQueueTable();
        metaMapper.createBindingTable();
        System.out.println("[DataBaseManager]创建表完成！");
    }

    //给数据库表中，添加默认数据
    //此处主要是添加一个默认的交换机
    private void createDefaultData() {
        //构造一个默认的交换机
        Exchange exchange=new Exchange();
        exchange.setName("");//匿名交换机（ RabbitMQ这样实现）
        exchange.setType(ExchangeType.DIRECT);
        exchange.setDurable(true);
        exchange.setAutoDelete(false);
        metaMapper.insertExchange(exchange);
        System.out.println("[DataBaseManager]创建初始数据完成！");
    }

    //把其他的数据库的操作，也在这个类中封装一下
    public void insertExchange(Exchange  exchange){
        metaMapper.insertExchange(exchange);
    }
    public List<Exchange> selectAllExchanges(){
        return metaMapper.selectAllExchanges();
    }
    public void deleteExchange(String exchangeName){
        metaMapper.deleteExchange(exchangeName);
    }
    public void insertQueue(MSGQueue queue){
        metaMapper.insertQueue(queue);
    }
    public List<MSGQueue> selectAllQueues(){
        return metaMapper.selectAllQueues();
    }
    public void deleteQueue(String queueName){
        metaMapper.deleteQueue(queueName);
    }
    public void insertBinding(Binding binding){
        metaMapper.insertBinding(binding);
    }
    public List<Binding> selectAllBindings(){
        return metaMapper.selectAllBindings();
    }
    public void deleteBinding(Binding binding){
        metaMapper.deleteBinding(binding);
    }

}
